package edu.cornell.cs4740.qa.ner;

import java.util.List;

import org.junit.Test;

import edu.cornell.cs4740.qa.WordContainer;
import edu.cornell.cs4740.qa.parsing.TokenFactory;

public class NamedEntityRecognizerTest {

	@Test
	public void test() {
		TokenFactory tokenFactory = new TokenFactory();
		List<WordContainer> tokenList = tokenFactory.sentenceToTokens("Hello, my name is Timothy D. Wing and I am awesome at 7:00PM");
		
		for(WordContainer wc : tokenList) {
			System.out.println(wc.getWord());
		}
		
		
		NamedEntityRecognizer ner = new NamedEntityRecognizer();
		System.out.println(ner.findNamedEntities("Hello, my name is Timothy D. Wing and I am awesome at 7:00PM"));
		ner.enrichWordContainersWithNamedEntities(tokenList);
	}
	
	@Test
	public void nerTest() {
		String s = "Hello, my name is Timothy D. Wing and I am awesome at 7:00PM";
		
		NamedEntityRecognizer ner = new NamedEntityRecognizer();
		ner.getNamedEntityMap(s);
	}

}
